# 1. 告诉Python我们要用的工具
import pandas as pd
import folium

# 2. 读取你的船舶数据文件
df = pd.read_csv(r'C:\Users\zhaoweizhen\Desktop\大学\AIS_2024_01_01.csv')

# 看看数据长什么样
print("数据读取成功！前5行数据如下：")
print(df.head())

# 3. 检查经纬度列名是否正确
print("\n数据的所有列名：")
print(df.columns.tolist())

# 4. 创建一个地图，中心点设在数据的平均经纬度上
map_center = [df['LAT'].mean(), df['LON'].mean()]
m = folium.Map(location=map_center, zoom_start=4)

# 5. 把每一艘船加到地图上（先只加前10艘）
for index, row in df.head(10).iterrows():
    # 检查经纬度是否有有效值
    if pd.notna(row['LAT']) and pd.notna(row['LON']):
        folium.Marker(
            location=[row['LAT'], row['LON']],
            popup=f"船名: {row.get('VesselName', '未知')}<br>速度: {row.get('SOG', 'N/A')}节"
        ).add_to(m)

# 6. 把地图保存成一个网页文件
m.save('我的第一张船舶地图.html')
print("完成！请用浏览器打开生成的 '我的第一张船舶地图.html' 文件！")